# 利用字典的特性编写一个通讯录程序


def address_book():
    print('|--- 欢迎进入通讯录程序 ---|')
    print('|--- 1：查询联系人资料  ---|')
    print('|--- 2：插入新的联系人  ---|')
    print('|--- 3：删除已有联系人  ---|')
    print('|--- 4：退出通讯录程序  ---|')
    contacts = dict()
    while 1:
        instr = int(input("请输入相关的指令代码："))
        if instr == 1:
            name = input("请输入要查询联系人姓名：")
            if name in contacts:
                print(name + ":" + contacts[name])
            else:
                print("您输入的姓名不在通讯录中！")
        if instr == 2:
            name = input("请输入要插入联系人姓名：")
            if name in contacts:
                print("您输入的姓名%s" % name + "已存在于通讯录中")
                if input("是否修改用户资料(YES/NO):") == "YES":
                    contacts[name] = input("请输入用户联系电话：")
                    print("插入用户信息成功！")
            else:
                contacts[name] = input("请输入用户联系电话：")
        if instr == 3:
            name = input("请输入要删除的联系人姓名：")
            if name in contacts:
                contacts.pop(name)
                print("删除用户信息成功！")
            else:
                print("您输入的联系人信息不存在！")
        if instr == 4:
            break
    print("-----感谢您使用通讯录程序-----")


if __name__ == "__main__":
    address_book()